package com.le.basic.subArrayProblem;

import org.junit.Test;

import java.util.HashMap;
import java.util.Map;

/**
 * 给定一个数组arr，和一个整数num，求在arr中，累加和等于num的最长子数组的长度
 * eg.
 * arr = {7,3,2,1,1,7,7,7}   num = 7
 * 其中有很多的子数组累加和等于7，但是最长的子数组是{3,2,1,1}，所以返回其长度4
 */
public class Code_01_LongestSumSubArrayLength {

    public static int maxLength(int[] arr, int k) {
        if (arr == null || arr.length == 0) {
            return 0;
        }
        // key : 0..i 累加和
        // value: 累加和出现最早的位置
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0, -1);
        int sum = 0;
        int len = 0;
        for (int i = 0; i < arr.length; i++) {
            sum += arr[i];
            // 0...i 累加和为sum
            // 0....k...i :0..k 为sum - k 则 k+1 ..i 是累加和为k的位置
            if (map.containsKey(sum - k)) {
                len = Math.max(i - map.get(sum - k), len);
            }
            if (!map.containsKey(sum)) {
                map.put(sum, i);
            }
        }
        return len;
    }

    @Test
    public void test() {
        int[] array = {7, 3, 2, 1, 1, 7, 7, 7};
        System.out.println(maxLength(array, 7));
    }
}
